#记得每次运行前修改dataSet路径和yolov8x-cls.yaml文件配置
from ultralytics import YOLO

# 加载模型
model = YOLO('yolov8x-cls.yaml')  # 从YAML构建新模型
model = YOLO('yolov8x-cls.pt')  # 加载预训练模型（推荐用于训练）
model = YOLO('yolov8x-cls.yaml').load('yolov8x-cls.pt')  # 从YAML构建并转移权重

# 训练模型，记得dataSet里面得有train和val文件夹
results = model.train(data='./dataSet_good3', epochs=10, imgsz=64)